'''
* This is the projet for Brtc LlmOps Platform
* @Author Leon-liao <liaosiliang@alltman.com>
* @Description //TODO 
* @File: 8_test_out.py
* @Time: 2025/10/22
* @All Rights Reserve By Brtc
'''
from datetime import datetime

import dotenv
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

dotenv.load_dotenv()
#1、编排 prompt
prompt = ChatPromptTemplate.from_messages([
    ("system", """  
      ## 角色
      请注意你是我的问题解决机器人， 你需要按照我的要求回答问题
      ## 要求
      - 直接给出问题的答案不要废话
      - 必要时请参考示例
      ## 示例
      Human:请问1+1等于几
      AI:2
      请仔细思考我的要求， 并直接给出答案，不要废话。
      """),
    ("human","{query}")
]).partial(now = datetime.now)

#2、构建大模型
llm = ChatOpenAI(model = "gpt-4o-mini")

print(llm.invoke(prompt.invoke({"query":"告诉我3个动物的名字。返回示例：1.狮子\n2.大熊猫"})))